package 每日一题;

/**
 * 1641. 统计字典序元音字符串的数目
 * 给你一个整数 n，请返回长度为 n 、仅由元音 (a, e, i, o, u) 组成且按 字典序排列 的字符串数量。
 *
 * 字符串 s 按 字典序排列 需要满足：对于所有有效的 i，s[i] 在字母表中的位置总是与 s[i+1] 相同或在 s[i+1] 之前。
 *
 *
 *
 * 示例 1：
 *
 * 输入：n = 1
 * 输出：5
 * 解释：仅由元音组成的 5 个字典序字符串为 ["a","e","i","o","u"]
 * 示例 2：

 * 输入：n = 2
 * 输出：15
 * 解释：仅由元音组成的 15 个字典序字符串为
 * ["aa","ae","ai","ao","au","ee","ei","eo","eu","ii","io","iu","oo","ou","uu"]
 * 注意，"ea" 不是符合题意的字符串，因为 'e' 在字母表中的位置比 'a' 靠后
 * 示例 3：

 * 输入：n = 33
 * 输出：66045
 * 提示：

 * 1 <= n <= 50
 */
public class D230329_T1641 {

    //以a 后面会有5种可能 e后面会有4种可能  i有3种 o有2种 u只有一种
    //状态转移方程  f(n) = {  f[*][1] = 1;
    //                   {  f[a][n] = f[e][n-1] + f[i][n-1] + f[o][n-1]+ f[u][n-1];
    public int countVowelStrings(int n) {

        return withCharEndNum('a',n)+withCharEndNum('e',n)+withCharEndNum('i',n)+withCharEndNum('o',n)+withCharEndNum('u',n);

    }

    /**
     * 末尾为当前字符 后面有多少种状态
     * @param c 特定字符 a e i o u
     * @param n 字符串剩余长度
     * @return 所有状态数量
     */
    private int withCharEndNum(char c,int n){

        if (n==1){
            return 1;
        }
        if (c=='a'){
            return withCharEndNum('a',n-1)+withCharEndNum('e',n-1)+withCharEndNum('i',n-1)+withCharEndNum('o',n-1)+1;
        }
        if (c=='e'){
            return withCharEndNum('e',n-1)+withCharEndNum('i',n-1)+withCharEndNum('o',n-1)+1;
        }
        if (c=='i'){
            return withCharEndNum('i',n-1)+withCharEndNum('o',n-1)+1;
        }
        if (c=='o'){
            return withCharEndNum('o',n-1)+1;
        }
        return 1;
    }


}
